

package com.hazelcast.jet.impl.client.protocol.task;

import com.hazelcast.client.impl.protocol.ClientMessage;
import com.hazelcast.client.impl.protocol.codec.JetTerminateJobCodec;
import com.hazelcast.instance.impl.Node;
import com.hazelcast.internal.nio.Connection;
import com.hazelcast.jet.impl.TerminationMode;
import com.hazelcast.jet.impl.TerminationMode.ActionAfterTerminate;
import com.hazelcast.jet.impl.operation.TerminateJobOperation;
import com.hazelcast.security.permission.ActionConstants;
import com.hazelcast.spi.impl.operationservice.Operation;

import javax.annotation.Nullable;
import java.util.UUID;

public class JetTerminateJobMessageTask extends AbstractJetMessageTask<JetTerminateJobCodec.RequestParameters, Void> {

    JetTerminateJobMessageTask(ClientMessage clientMessage, Node node, Connection connection) {
        super(clientMessage, node, connection, JetTerminateJobCodec::decodeRequest, o -> JetTerminateJobCodec.encodeResponse());
    }

    @Override
    protected UUID getLightJobCoordinator() {
        return parameters.lightJobCoordinator;
    }

    @Override
    protected Operation prepareOperation() {
        return new TerminateJobOperation(parameters.jobId, TerminationMode.values()[parameters.terminateMode], parameters.lightJobCoordinator != null);
    }

    @Override
    public String getMethodName() {
        return "terminateJob";
    }

    @Override
    public Object[] getParameters() {
        return new Object[]{};
    }

    @Nullable
    @Override
    public String[] actions() {
        ActionAfterTerminate terminationMode = TerminationMode.values()[parameters.terminateMode].actionAfterTerminate();
        switch (terminationMode) {
        case RESTART:
        case SUSPEND:
            return new String[]{ActionConstants.ACTION_RESTART};
        case CANCEL:
            return new String[]{ActionConstants.ACTION_CANCEL};
        default:
            throw new IllegalArgumentException("Unknown action after termination mode " + terminationMode);
        }
    }
}
